import pyE17
from scipy import ndimage
import matplotlib.pyplot as plt

p = pyE17.utils.imload('pinhole.png')
# Let's try this: same probe, no propagation
#probe = pyE17.prop.free_nf(p, 2e-10, .7e-3, 20e-9)

# ... But on a smaller array
probe = ndimage.zoom(p, 192./256)[32:-32,32:-32]
plot = plt.imsave('this.png',probe)
#pyE17.io.h5write('probe_1_128.h5', probe=probe) 
